# 把png文件处理成不会有控制台警告的样子
# 在Photoshop没有这个选项

from PIL import Image
import os

def fix_png_icc_profile(input_path, output_path):
    with Image.open(input_path) as img:
        # 移除 ICC 配置文件
        if 'icc_profile' in img.info:
            img.info.pop('icc_profile')
        # 保存图片（使用 PNG 格式，不保留配置）
        img.save(output_path, 'PNG')

if __name__ == '__main__':
    # 处理当前文件夹下的所有PNG 图片
    for filename in os.listdir('.'):
        if filename.endswith('.png'):
            input_path = os.path.join('.', filename)
            output_path = os.path.join('.', filename)
            fix_png_icc_profile(input_path, output_path)
    